import sys
import time

from PyQt6.QtWidgets import QApplication, QWidget, QPushButton, QGridLayout
from PyQt6.QtGui import QIcon  # 导入引用图标的函数
from baiduWenkuDownload import wenku


class MainWindow(QWidget):
    # 主窗口初始化
    def __init__(self):
        super().__init__()
        self.setGeometry(200, 200, 800, 400)  # 设置初始位置与窗口大小
        self.setWindowIcon(QIcon("./tool.jpg"))  # 加载窗口图标
        self.setWindowTitle('工具包')
        self.setStyleSheet('background-color:#FBEAFF')  # 设置窗口内背景颜色
        self.initUI() # 初始化界面
        self.show() # 显示窗口


    def initUI(self):   #创建一个5*5的数组，后面会逐一放入栅格
        keys = [
            ['百度文库下载', '待开发', '待开发', '待开发'],
            ['待开发',  '待开发',  '待开发',  '待开发'],
            ['待开发', '待开发', '待开发', '待开发'],
            ['待开发', '待开发', '待开发', '待开发'],
            ['待开发', '待开发', '待开发', '待开发'],
        ]
        grid = QGridLayout() # 创建网格布局管理器
        self.setLayout(grid) # 将网格布局管理器应用到窗口

        bdButton = QPushButton(keys[0][0],self)
        bdButton.setStyleSheet('background-color:#D3BCF2')
        bdButton.setFixedSize(100, 50)
        bdButton.clicked.connect(self.goWenku)
        grid.addWidget(bdButton, 0, 0)


    # 跳转到文库
    def goWenku(self):
        self.subWindow = wenku.MyWindow()
        self.subWindow.show()



if __name__ == '__main__':
    app = QApplication(sys.argv)
    win = MainWindow()
    sys.exit(app.exec())
